from .prepare_picture import Classification

datasets = {
    'normal': Classification,
}


def get_classification_dataset(name, **kwargs):
    return datasets[name.lower()](**kwargs)
